from torchsummary import summary
# from mono_fm.depth_encoder import DepthEncoder
from mono_fm.depth_decoder import DepthDecoder
import torch

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = DepthDecoder(6)
model.to(device)
print(model)
summary(model, [(64, 320, 96), (256, 80, 24), (512, 40, 12), (1024, 20, 6), (2048, 10, 3)])